package 简单.字符串;

/**
 * 句子 是一个单词列表，列表中的单词之间用单个空格隔开，且不存在前导或尾随空格。每个单词仅由大小写英文字母组成（不含标点符号）。
 * <p>
 * 例如，"Hello World"、"HELLO" 和 "hello world hello world" 都是句子。
 * 给你一个句子 s​​​​​​ 和一个整数 k​​​​​​ ，请你将 s​​ 截断 ​，​​​使截断后的句子仅含 前 k​​​​​​ 个单词。返回 截断 s​​​​​​ 后得到的句子。
 * <p>
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode-cn.com/problems/truncate-sentence
 */
public class 截断句子_1816 {

    public static void main(String[] args) {

        String str = "Hello how are you Contestant";
        System.out.println(truncateSentence(str, 4));

    }

    /**
     * 找到第k个空格的索引
     *
     * @param s
     * @param k
     * @return
     */
    public static String truncateSentence(String s, int k) {
        int index = 0;
        int cur = 0;
        for (int i = 0; i < s.length(); i++) {
            if (s.charAt(i) == ' ') {
                cur++;
            }
            index = i;
            if (k == cur) {
                break;
            }
        }
        if (index == s.length() - 1) {
            return s;
        }
        return s.substring(0, index);
    }

}
